package hotel.base.oversea.utils;

import org.springframework.cglib.beans.BeanCopier;

import java.util.WeakHashMap;

/**
 * @author HuangT
 * @Date 2020/8/18 10:46
 */
public class BeanCopierUtil {

    /**
     * BeanCopier的缓存
     */
    static final WeakHashMap<String, BeanCopier> BEAN_COPIER_CACHE = new WeakHashMap<>();

    /**
     * BeanCopier的copy
     *
     * @param source 源文件的
     * @param target 目标文件
     */
    public static void copy(Object source, Object target) {
        if (null == source) {
            throw new NullPointerException("拷贝对象为空");
        }
        String key = genKey(source.getClass(), target.getClass(), "0");
        BeanCopier beanCopier;
        if (BEAN_COPIER_CACHE.containsKey(key)) {
            beanCopier = BEAN_COPIER_CACHE.get(key);
        } else {
            beanCopier = BeanCopier.create(source.getClass(), target.getClass(), false);
            BEAN_COPIER_CACHE.put(key, beanCopier);
        }
        beanCopier.copy(source, target, null);
    }

    /**
     * converterCopy
     * 属性不同时,也可以复制,前提要先定义好属性
     *
     * @param source 源文件的
     * @param target 目标文件
     */
    public static void converterCopy(Object source, Object target) {
        String key = genKey(source.getClass(), target.getClass(), "1");
        BeanCopier beanCopier;
        if (BEAN_COPIER_CACHE.containsKey(key)) {
            beanCopier = BEAN_COPIER_CACHE.get(key);
        } else {
            beanCopier = BeanCopier.create(source.getClass(), target.getClass(), true);
            BEAN_COPIER_CACHE.put(key, beanCopier);
        }
        /*
         目前只适用将所有的Long转成int,如果需要其他属性转换,则按格式自行添加
         一种类型只能对应转成另一种其他类型,不能同一个对象中有的Long形想转成int,有的想转成String,这样会报错
         遇到这种情况时,可将属性名称改成不同名称,然后使用get set手动设置即可
         */
        beanCopier.copy(source, target, (sourceValue, targetClass, methodName) -> {
            if (sourceValue instanceof Long) {
                return ((Long) sourceValue).intValue();
            }
            return sourceValue;
        });
    }


    /**
     * 生成key
     *
     * @param srcClazz 源文件的class
     * @param tgtClazz 目标文件的class
     * @return string
     */
    private static String genKey(Class<?> srcClazz, Class<?> tgtClazz, String type) {
        return srcClazz.getName() + tgtClazz.getName() + type;
    }


}
